package huawei;

/**
 * @author pengfei.hpf
 * @date 2020/3/26
 * @verdion 1.0.0
 * 1109. 航班预订统计
 * 这里有 n 个航班，它们分别从 1 到 n 进行编号。
 *
 * 我们这儿有一份航班预订表，表中第 i 条预订记录 bookings[i] = [i, j, k] 意味着我们在从 i 到 j 的每个航班上预订了 k 个座位。
 *
 * 请你返回一个长度为 n 的数组 answer，按航班编号顺序返回每个航班上预订的座位数。
 *
 *
 *
 * 示例：
 *
 * 输入：bookings = [[1,2,10],[2,3,20],[2,5,25]], n = 5
 * 输出：[10,55,45,25,25]
 *
 *
 * 提示：
 *
 * 1 <= bookings.length <= 20000
 * 1 <= bookings[i][0] <= bookings[i][1] <= n <= 20000
 * 1 <= bookings[i][2] <= 10000
 */
public class CorpFlightBookings {

    //TODO: 模拟公交车上车, 下车, 1->2 有10个人, 那么意味着第三站就要下10个人. 最后每站有多少人就是前面没下的人加起来的总数
    public int[] corpFlightBookings(int[][] bookings, int n) {
        int[] res = new int[n];
        if(bookings == null || bookings.length == 0 || bookings[0].length == 0){
            return res;
        }
        for(int[] book: bookings){
            int start = book[0];
            int end = book[1];
            int count = book[2];
            res[start - 1] += count;
            if(end < n){
                res[end] -= count;
            }
        }
        for(int i = 1; i < n; i ++){
            res[i]+= res[i - 1];
        }
        return res;
    }
}
